/*
 * Copyright (C) 2012 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.android.keyguard;

import static com.android.systemui.DejankUtils.whitelistIpcs;

import android.app.admin.DevicePolicyManager;
import android.content.Context;
import android.content.res.Resources;
import android.telephony.SubscriptionManager;
import android.telephony.TelephonyManager;
import android.util.Log;

import com.android.internal.widget.LockPatternUtils;
import com.android.keyguard.KeyguardSecurityModel.SecurityMode;
import com.android.systemui.Dependency;
import com.unisoc.keyguard.util.SimLockUtil;
import com.unisoc.keyguard.KeyguardSimLockMonitor;
import com.unisoc.sdk.common.telephony.UniTelephonyManager;

import javax.inject.Inject;
import javax.inject.Singleton;

@Singleton
public class KeyguardSecurityModel {

    /**
     * The different types of security available.
     * @see KeyguardSecurityContainer#showSecurityScreen
     */
    public enum SecurityMode {
        Invalid, // NULL state
        None, // No security enabled
        Pattern, // Unlock by drawing a pattern.
        Password, // Unlock by entering an alphanumeric password
        PIN, // Strictly numeric password
        SimPin, // Unlock by entering a sim pin.
        SimPuk, // Unlock by entering a sim puk
        SimLock, // Unisoc: Support SimLock
        /* For SubsidyLock feature @{ */
        SubsidyLock_Lock, // Lock UI for SubsidyLock
        SubsidyLock_EnterCode, // Local unlock UI for SubsidyLock
        SubsidyLock_Init //Init UI for SubsidyLock
        /* @} */
    }

    private final static String TAG = "KeyguardSecurityModel";
    private final Context mContext;
    private final boolean mIsPukScreenAvailable;

    private LockPatternUtils mLockPatternUtils;

    @Inject
    KeyguardSecurityModel(Context context) {
        mContext = context;
        mLockPatternUtils = new LockPatternUtils(context);
        mIsPukScreenAvailable = mContext.getResources().getBoolean(
                com.android.internal.R.bool.config_enable_puk_unlock_screen);
    }

    void setLockPatternUtils(LockPatternUtils utils) {
        mLockPatternUtils = utils;
    }

    public SecurityMode getSecurityMode(int userId) {
        KeyguardUpdateMonitor monitor = Dependency.get(KeyguardUpdateMonitor.class);

        if (mIsPukScreenAvailable && SubscriptionManager.isValidSubscriptionId(
                monitor.getNextSubIdForState(TelephonyManager.SIM_STATE_PUK_REQUIRED))) {
            return SecurityMode.SimPuk;
        }

        if (SubscriptionManager.isValidSubscriptionId(
                monitor.getNextSubIdForState(TelephonyManager.SIM_STATE_PIN_REQUIRED))) {
            return SecurityMode.SimPin;
        }

        /* Unisoc: Support simlock @{ */
        if ((SimLockUtil.isAutoShow() && (SubscriptionManager.isValidSlotIndex(
                monitor.getNextSlotIdForSimLockState(TelephonyManager.SIM_STATE_NETWORK_LOCKED))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_NETWORKSUBSET_LOCKED))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_SERVICEPROVIDER_LOCKED))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_CORPORATE_LOCKED))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_SIM_LOCKED))
                ||  SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_NETWORK_LOCKED_PUK))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_NETWORK_SUBSET_LOCKED_PUK))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_SERVICE_PROVIDER_LOCKED_PUK))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_CORPORATE_LOCKED_PUK))
                || SubscriptionManager.isValidSlotIndex(
                        monitor.getNextSlotIdForSimLockState(UniTelephonyManager.SIM_STATE_SIM_LOCKED_PUK))
                )) || KeyguardSimLockMonitor.isSimLockStateByUser()
                || SimLockUtil.getInstance(mContext).showSimLockWhenSimsAbsent(mContext)) {
            if (!KeyguardSimLockMonitor.getSimLockCanceled()) {
                Log.d(TAG,"getSecurityMode: return SimLock");
                return SecurityMode.SimLock;
            } else {
                Log.d(TAG,"getSecurityMode: SimLock already canceled by user!");
            }
        }
        /* @} */

        /* For SubsidyLock feature @{ */
        KeyguardSubsidyLockController subsidyController = KeyguardSubsidyLockController.getInstance(mContext);
        if (subsidyController.isSubsidyLock()) {
            if(subsidyController.isSubsidyEnterCode()){
                return SecurityMode.SubsidyLock_EnterCode;
            } else if(subsidyController.getSubsidyLockMode() ==  KeyguardSubsidyLockController.SUBSIDY_LOCK_SCREEN_MODE_LOCK ||
                    subsidyController.getSubsidyLockMode() ==  KeyguardSubsidyLockController.SUBSIDY_LOCK_SCREEN_MODE_SWITCH_SIM){
                return SecurityMode.SubsidyLock_Lock;
            }
        }
        /* @} */

        final int security = whitelistIpcs(() ->
                mLockPatternUtils.getActivePasswordQuality(userId));
        switch (security) {
            case DevicePolicyManager.PASSWORD_QUALITY_NUMERIC:
            case DevicePolicyManager.PASSWORD_QUALITY_NUMERIC_COMPLEX:
                return SecurityMode.PIN;

            case DevicePolicyManager.PASSWORD_QUALITY_ALPHABETIC:
            case DevicePolicyManager.PASSWORD_QUALITY_ALPHANUMERIC:
            case DevicePolicyManager.PASSWORD_QUALITY_COMPLEX:
            case DevicePolicyManager.PASSWORD_QUALITY_MANAGED:
                return SecurityMode.Password;

            case DevicePolicyManager.PASSWORD_QUALITY_SOMETHING:
                return SecurityMode.Pattern;
            case DevicePolicyManager.PASSWORD_QUALITY_UNSPECIFIED:
                /* For SubsidyLock feature @{ */
                if (Resources.getSystem().getBoolean(com.android.internal.R.bool.config_subsidyLock)
                        && subsidyController.getSubsidyLockMode() == KeyguardSubsidyLockController.SUBSIDY_LOCK_SCREEN_MODE_INIT) {
                    return SecurityMode.SubsidyLock_Init;
                }
                /* @} */
                return SecurityMode.None;

            default:
                throw new IllegalStateException("Unknown security quality:" + security);
        }
    }
}
